package com.zwj.interview.BFS;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @Author: Zwj
 * @Date: 2022/9/19 15:57
 */
public class 二叉树的层平均值 {

    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> result=new ArrayList<>();
        if(root==null){
            return result;
        }
        //这个queue保存着每一层的节点
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
           int size=queue.size();
           double sum=0;
           for(int i=0;i<size;i++){
               TreeNode cur = queue.poll();
               sum+=cur.val;
               if(cur.left!=null){
                   queue.add(cur.left);
               }
               if(cur.right!=null){
                   queue.add(cur.right);
               }
           }
           result.add(sum/size);
        }
        return result;
    }




}